Skip to main content

Setting up Agents to query your MongoDB Database

This guide will walk you through configuring Agents to query your MongoDB Database. Before following this guide, ensure that you have set up your MongoDB tools already by following the instructions here.

To configure these agents, you will require the following components added to your project

  1. MongoDB Find Tool
  2. MongoDB Aggregate Tool
  3. MongoDB Metadata Tool
  4. Memory
  5. LLM model

Metadata Retrieval Agent​

The Metadata Retrieval Agent will be a scheduled Agent to retrieve the Metadata associated with the MongoDB Database, and populate a Memory object.

SQL Query Agent​

The SQL Query Agent will use the Metadata retrieved from the Memory object to allow the user to query their Database.

Agent Setups​

Metadata Retrieval Agent​

The Agent setup should have a LLM, Memory block, and Metadata Retrieval tool.

MongoDB Metadata agent

  1. Add the LLM to the Agent
  2. Add the Metadata Retrieval tool to the LLM
  3. Add the Memory block to the Agent

The following connections are needed:

  1. Input line to LLM
  2. LLM line to Output
  3. Memory line to Output

Schedule the Agent by

  1. Selecting the gear icon on the Input block
  2. Turning on Schedule Execution
  3. Providing a time range to repeat metadata retrieval. This would be based on how often you expect your data to change.

Add MongoDB Query tool

Query Agent​

The Agent setup should have a LLM, Memory block (same as previous), and SQL Query tool. MongoDB Query agent

  1. Add the LLM to the Agent
  2. Add the MongoDB Find tool to the LLM
  3. Add the MongoDB Aggregate tool to the LLM
  4. Add the Memory block to the Agent

The following connections are needed:

  1. Input line to LLM
  2. LLM line to Output
  3. Memory line to LLM

Query the LLM by asking it questions relevant to your Database. Ensure that the Metadata agent has been run at least once before running queries, as it is required to populate any information about the database in the Memory block.

MySQL query results

[!NOTE] Your MongoDB server must be accessible by Public IP. If you're having trouble connecting to your MongoDB server, you can reach out to us at [email protected] and we can provide you any IPs you may need to whitelist on your firewall to allow the tools to connect.